CSS transition does not work when I toggle the class with JS